package main.java.DMDE.infrastructure.visual;

import javax.swing.tree.DefaultMutableTreeNode;

import main.java.DMDE.model.Attribute;

public class AttributeTreeNode extends DefaultMutableTreeNode  {

    private static final long serialVersionUID = 1L;
    private Attribute attribute;
    
    public AttributeTreeNode(Attribute attribute){
	this.attribute = attribute;
    }
    
    public Attribute getAttribute() {
	return attribute;
    }
    
    @Override
    public String toString() {
        return attribute.getName();
    }

}
